Update Neural_net_test.py