some more updates to script