Add tests for word count.