Update NeuralNet.py