Update neural_net.erl