updated data set and training algorithms