modified tester to do stratified k-fold CV