Added functionality to restart game.