def METHOD_NAME(hist, **kwargs):
    """
    This extra line allows to set a fixed colormap range when drawing the
    plot 2D for the first time, when inserted in the right place in the original
    code.
    """
    hist.setLevels(*(kwargs["zrange"] if "zrange" in kwargs else hist.getLevels()))
    return None