Added partition_classes function