updated train_grid

now extracting unigrams and bigrams