added first testloss plot