Added a ! to train function to indicate that modifications to Jordan RNN will be done in-place.