slight change to script