some nice edits here and there, mainly with respect to multiclass classification settings