Plotting out training and testing splits