more small changes to script