Finishing assigment
Finished all coding