#1 fixed

Updated code to work with new site layout