Update script

Should be complete