Better plotting of learning progress.