Add init code for Chapter 2