Update NeuralNetwork.scala