Problem 4 rewritten