Added support for game states and a basic main menu