Reduced the test dataset