Add restore command