Implement feature to restart game with a button click.