Add tf-idf to the code