Added the neural network code.