updates on main.py