wrote test with smaller dataset