[Debug] moved data loading part out of plot_dataset