Finished part 1 homework