Added unit tests for wordcount.py