Slight docstring update