Update main.py

From Luming