update comments in main.py