Update main.py

Still working on restore function...