add db restore function