create_dataset function

assign the classes to the dataset list