1. add logistic regression