Added Figure 5.22

Also applied pep8