now using training set