Updated

Completed the code and assignment