update main __init__.py