Moved update script to restart section