Added main.py file