add batch train loss graph
big fuckery of plotting