Adding the neural network classifier