Edited comments in main.py