Add Chapter 4 code