Update load-dataset.R