main.py: better comments