add /restore support