Changed load_dataset to loadDataset to maintain the naming convention followed in the rest of the code