Minimal, but still interesting change to the code.
Alterations have been reflected in chapter 2.