Improving the update script