Slight change on NeuralNetwork interface