Revert "Update main.py"

This reverts commit e7d8e130f62a89ffc8f1b89537d6b332e0549627.