Completed problem 4.1