Changed to a bag of words model