Updated menu and game functions