'Added ability to restart game'