small changes to script