Enhance update script