Minor modification in MulticlassModel