Update main.py

fixed description