Updated game logic