minor edits to script