cleaned the wordcount.py demo