Use python docstrings for docs