Updated dataset_cleanup.r