Update main.py

fix bug