Modify Problem 4