code: def dispersion(words, corpus, y=None, ax=None, colors=None, colormap=None,
                 labels=None, annotate_docs=False, ignore_case=False, **kwargs):
      visualizer = DispersionPlot(
          words, ax=ax, colors=colors, colormap=colormap,
          ignore_case=ignore_case, labels=labels,
          annotate_docs=annotate_docs, **kwargs
      )
      visualizer.fit(corpus, y, **kwargs)
      return visualizer.ax