Continued working on code

Continued working on code