changing fire_dataset to load_dataset