Add NeuralNet base class