Update script improved