Add docstrings to __init__.py