Add restoreDb method