Update word_histogram.py