eliminated load_data function, data_generator becomes load_data