Problem 4 update