Fixed bugs with restarting the game after victory and game over