Restoring main.py