PEP8 and docstring fix