added more game logic