Added game state management