Minor changes to script.