pep8, docstring, add a test