update test_NB method