minor changes to script