update fortran obj example to run multiple steps

example/ALL_test_f_obj.F90

@@ -50,7 +50,7 @@ program ALL_test_f
     integer                             ::  error
     integer                             ::  rank, n_ranks
     real(8),dimension(:,:),allocatable  ::  new_vertices
-    integer                             ::  i
+    integer                             ::  i,j
 
     type(ALL_t)     ::  lb
 
@@ -82,15 +82,17 @@ program ALL_test_f
         end if
     end do
 
+    ! set up balancer
     call lb%init(ALL_STAGGERED, 3, 4.0d0)
     call lb%set_proc_tag(rank)
-    call lb%set_work(real( product(coords,1)*64,8))
     call lb%set_vertices(vertices)
-    ! if using a non cartesian communicator, this call would be required
+    ! if using a non cartesian communicator, this call would be required instead
     ! call lb%set_proc_grid_params(coords,dims)
     call lb%set_communicator(cart_comm)
-
     call lb%setup()
+
+    do j = 1, 2
+        call lb%set_work(real( product(coords,1)*64,8))
         call lb%balance()
 
         ! If not allocatable as target, must use
@@ -106,6 +108,7 @@ program ALL_test_f
                 call MPI_BARRIER(cart_comm, error)
             end if
         end do
+    end do
     deallocate(new_vertices)
     call MPI_FINALIZE(error);