def forceSquare(ax):
    """ Force axes to be square regardless of data limits """
    if ax is not None:
        x0, x1 = ax.get_xlim()
        y0, y1 = ax.get_ylim()
        ax.set_aspect((x1 - x0) / (y1 - y0))
