PEP8 fixes for Chapter 2