Add Restart Functionality.