Problem 3 revision