Adding restore command