Added a bit more game logic