Wrote test for count_words