Problem 1 improved