update QueryLikelihood Model code