Added some more game logic