PEP8 and docstrings cleanup