Minor changes to script