Add chapter 2 codes