Extracted TrainCommandOutputProcessor to own file and created unit test