Problem 5 improved