finish up word2vec improvements