Learn to document more. It gets a little better with age once you must resign yourself to the fact that you will be interrupted at any point. If you document, you can resume easier and there’s less mind shift inertia.