more improvements to multiclass classification

see line 70. assign might be a better word.