Correct docstring and ensure PEP8 compliance.