Adding bag of words part