Update main.py

I am exploring this option