Modify kneser ney calculation