added a button to restart the app