Added both training/testing set as what we are evaluating CV 10-fold on, some more clean up