minor updates to script