Update main.py

log