Test with smaller dataset