1 files changed, 6 insertions, 1 deletions

diff --git a/dis/original/Transitive/transitive.c b/dis/original/Transitive/transitive.c
index 854d57c..ca16a31 100644
--- a/dis/original/Transitive/transitive.c
+++ b/dis/original/Transitive/transitive.c
@@ -18,6 +18,7 @@
 #include <assert.h>
 #include <stdlib.h>
 #include "DISstressmarkRNG.h"
+#include "extra.h"
 
 #define MIN_VERTICES 8
 #define MAX_VERTICES 16384
@@ -34,7 +35,7 @@
  * main()
  */
 
-int main(){
+int main(int argc, char** argv){
   unsigned int *din, *dout;
   unsigned int n;
   unsigned int m;
@@ -68,12 +69,14 @@ int main(){
     *(din + j*n + i) = randInt(MIN_EDGES, MAX_EDGES);
   }
   
+  SET_UP
   startTime = time(NULL);
 
   for (k=0; k<n; k++){
   unsigned int old;
   unsigned int new1;
   unsigned int *dtemp;
+    START_LOOP
     
     for (i=0; i<n; i++){
       for (j=0; j<n; j++){
@@ -87,6 +90,7 @@ int main(){
       dtemp = dout;
       dout = din;
       din = dtemp;
+      STOP_LOOP
     }
 
     startTime = time(NULL) - startTime;
@@ -112,6 +116,7 @@ int main(){
     fprintf(stdout, " total time = %u seconds. \n", (unsigned int)startTime);
     free(din);
     free(dout);
+    WRITE_TO_FILE
     return(0);
   }