Add some Chapter 2 code