added legend for logistic regression