small update to the script