Revert "Update main.py"

This reverts commit f4469ff9151f7140c4463a3b510aa06fb2f300f7.