problem 4 updated