Another update to the update script