Got count_words method working