upate main.py