pep8 + add a test