Add / fix some docstrings