Docs: docstring fixes