Complete for hw2 part 2