Code updated

Added the changes I have done to the site.