Update __init__.py

cleanup, update docstrings