Updated update script