Update script to run all the way