Adding docstring to word2vec