Add word_count.py