Super simple main.py code now