modified __init__.py