Added missing comment in main.py!