Add restore method with passing tests.