Updated the script