using training sets now not full data