Set your task. When you drift, hit the prompt — it asks you one sharp question and pulls you straight back to what you were doing.