added training set
todo:
1) change training set