Fixing WordCloud.
Updating word cloud code.