Small docstring improvements