Finished the assignment