return changed to raise
Added main.py
main.py added