Added neural network code to ANN.ipynb