pep8 tweaks
My solution wasn't perfect, but need to improve this one.