Update neural_network.py