still working on restore problem