Problem 2 update