Tweaks to the Chapter 2 notebook