added button to restart game