refine logit regression part