add loadDataset function