Added the main menu and a couple more enums for states