Change the restore files code

Used the simpler code from the restore script that prompts.