Update main.py

Changed back