Exercise 3.5 pep8 cleanup