Updating Task 2 waiting

src/ag_gen/ag_gen.cpp

@@ -485,7 +485,7 @@ AGGenInstance &AGGen::generate(bool batch_process, int batch_size, int numThrd,
                         std::cout << "Task 2 Finished." << std::endl;
                     }
                 }
-            }
+            
                 else{
                     /*
                     One, need to listen for Updates to instances.facts
@@ -544,9 +544,10 @@ AGGenInstance &AGGen::generate(bool batch_process, int batch_size, int numThrd,
                                 gettimeofday(&t32,NULL);
                                 total_task3+=(t32.tv_sec-t31.tv_sec)*1000.0+(t32.tv_usec-t31.tv_usec)/1000.0;
                             }
-                    }
-                }
+                        } //Node 0 Task 3 wait ends
+                    } //while probe/instance update ends
                     std::cout << "Node " << world.rank() << " has finished Task 2 waiting." << std::endl;
+                } //else wait ends
             }
 
             //Task Four