Create word_count.py