Updated the script to work with the new site