problem 4 small corrections