added working main.py file