added __init__.py to proper importing and main.py