Updated restore script