Tidy up
Smaller dataset