Small docstring fixes