Added tests for count_words