Finish hw4 part 1