improved training set