Change to kill zombie procs after loop

Assignment2/main.c

@@ -146,10 +146,9 @@ void processtree() {
             sprintf(num, "The value is %d\n", i);
             print_prompt(num);
             exit(0);
-        } else {
-            wait(NULL);
         }
     }
+    while (wait(NULL) > 0);
 }
 
 int main() {