Update WordCount.py

After more consideration, I think this one is better.