Add custom equality function as walk_together argument