updating Problem 3