Update bag_of_words.rb