Added more game logic