Improve a couple of docstrings