* Activation function is only kept in the Layer class now