Modifies the trainer classes
Adds plots