Finished the code

Finished the code