Minor fixes in NeuralNet