Modified the z_vertex_opt correction to do it iteratively