minor edit to load_data