Adding plot of training distance plot