added support for word2vec