fixed bugs

fixed bugs in NB and add training data