Handle weights (and number of dropped assignments), not sure if it works for all classes though.