completed test/train data split and added function to plot model performance over time