Updated documentation on how to train the neural network