Add force restore option