Plotting the training and testing results