Reworked the dataset_loader classes, still toying with benchmarking and result storing