training small data set complete

bugs fixed

things to do: testing, training with large data set