Clicking update button focuses code

This already worked, but now there's a test.

Fixes #39