Completed problem 2.1