Tweaking comments in main.py