Add comment on main.py