Changed n_labels to n_classes