Updated the word embedding pickler