Update main.py

einige änderungen