Adding smaller data set for testing