Added states for menu and game