Added code to restart application