Split out data set handling class