Word count is working perfectly. Word vector is greedy