Problem 5 updated