Implement wordcount.py assignment