Updated code for website changes

Updated code for changes to the website