Update main.py

Release V1.0.1