Improve some docstrings