Add class for tf-idf logic