added some images of training set. not finished