Update word_count.py