Add more game logic