Cleaned up word2vec script and added composition in it

Now assuming Python 3