Added training data to the neural network