Initial addition of main.py