Problem 3 updated