Adjust chapter 3 code block