Added ability to restart game.