Add restore.py