Smaller data set to test on