adding training set stuff