Yes, dicipline is needed there as well. I do a lot if genetic reseach and I look at my laptop as a tool, a potentially dangerous tool, like a woodworker using a lathe, i stay alert and focused on the task at hand.
My laptop is essentially a desktop as well. I only use it on a desk in one room.
I never turn to these devices out of boredum. That is the best rule.
Plus, steven black on github has a great host file blocking all kibds of websites.
