Working on logistic regression layer.