Revise solution to problem 3.