code: def finalize(self, **kwargs):
          
  
          self.ax.set_ylim(-1, len(self.indexed_words_))
          self.ax.set_title("Lexical Dispersion Plot")
          self.ax.set_xlabel("Word Offset")
          self.ax.grid(False)
  
          # Add the legend outside of the figure box.
          if not all(self.classes_ == np.array([self.NULL_CLASS])):
              box = self.ax.get_position()
              self.ax.set_position([box.x0, box.y0, box.width * 0.8, box.height])
              self.ax.legend(loc='center left', bbox_to_anchor=(1, 0.5))
docs: The finalize method executes any subclass-specific axes
          finalization steps. The user calls poof & poof calls finalize.
          Parameters
          ----------
          kwargs: generic keyword arguments.