Added implementation of new method in mock dataset