Modified chapter 4 figures to work under py 2.x or py 3.x

* Verified figures are the same as in book