Add basic TF-IDF needs integration w dataset though