Problem 4 Updated