Fix formatting (PEP8 compliant + docstrings)