Refactor docstrings to be more consistent