Using tf-idf to improve the result