add the implementation of the neural network