Improve problem 1.