Update to game logic
Fix