updata main.py