Fixes for TFIDF lab