Adding test for unigram function