Update docstrings to work better for sphinx/numpy autodoc.