Working on the train pipeline class.
Running the new prediction.