add WeightedClassesWrapper