Add some game logic