adding training set