add restore option

add restore option