Flow balance almost done