Cleaning up update script