Improvements to the update script