Added a boolean switch to make the logistic regression layer optional.