plot_improvement works OK, could use fine tuning

plot improvement is a mess, but getting there

Began work on plotting improvements between training runs