Add chapter 4 codes