update count_words.py