added function to plot staged MSE learning curves