Cleaned up the output of the trainer.