Refactor docstring to fit pep257