smaller dataset for testing