cleaning up the code

pep8 and flake8 are happy