minor edit to hw2 problem 2