Update main.py

moving forward on restore