Finish hw4 Part 2