Neural Network code
New NN code