Fixes and improvements to VisualizeCorpus.