got rid of debug statements
Now should do a couple runs for each training size and plots with error bars