Finish hw4 Part 1