Improved the menus a bit.

Cleaned up the key handling a bit too.

Need to figure out a way to save the game state in the future. :)