add trainNB