First hacks for the logistic regression layer