Update naivebayes_driver
1. fix memory leak