Modularize the whole screen switching more.

It was getting a real pain to move keep track of everything in their own
variables. This way, it should be easier to add more screens and such.