Update main.py

second commit from Luming