WIP add word2vec class features