Modified the interface of the class. Recovery works under 2-D case