Added some game logic