Updated with addition of gensim comparison stuff.