Revert "Update main.py"

This reverts commit 9f1d425ee01457dc7f3fd31a1258709f39c6064b.