updated models, added NB info gain