update __init__.py and documentation