clean up tests add pep8