Update NeuralNetwork.java