Update main.py

v1.0.5