updateFocus update

I actually like the new updateFocus() method in the main class. It's
rather elegant.

I'll probably keep it like this.