Docstring fixes