Add main.py file