Add button to restart