Fix bug. Multiclass classification.