Update Neural_net_converge.py