Fixed restarting the game