Added code for bigrams