Update main.py

update 4.1