Almost done with the restoring option