Fixed imports in main.py, deleted unnecessary __init__.py