added pep8 and a test