Simple implementation of the neural network