More game state logic in main