A little Pep 8 refactoring...