Nearly done with exercise 3.1