Slight docstring improvements