Update main.py

test description