Update main.py

Update comment