Add test for Word2Vec algorithm, minor fixes