add main.py

added main python file