clean code to make pep8 compliant