Update script improvements