Update NeuralNetwork.py