Update script with new changes