added main code back to __init__.py