add part of multi-class logistic regression