Add restore option