Updated update script slightly