Worked on code

Worked on code