Finished Part 1/2 of the assignment