Modified SVMMain to only do testing on train (for re-plotting purposes)
Added parameters and functionality to plot_learning_curves