minor edit to problem 4