updating docs and __init__.py