Focus on the console after running the code.

Addresses #8.