Added ability to restart the app