added training set