more updates to the script