Updated the script again