Improve some docstrings in util