Next step, simplifying layers. Added unit tests for logistic regression and MLP