add more trainning data
bug fix