Fix class docstrings