Actually have the restore command working