fix docstring + PEP8