modify ngram_utils.py