Adding code to chapter 02