fixed plot Test Train so I can generate good figures

I am not convinced that the training error looks correct