Code updated

I have added some elements and code.