cleanup code for pep8