Clean up problem 2.