Waiting for processes to finish (using the wait function) in C

main.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/wait.h>

int main(int argc, char* argv[]) {
    int id = fork();
    int n;
    if (id == 0) {
        n = 1;
    } else {
        n = 6;
    }
    
    if (id != 0) {
        wait(NULL);
    }
    
    int i;
    for (i = n; i < n + 5; i++) {
        printf("%d ", i);
        fflush(stdout);
    }
    if (id != 0) {
        printf("\n");
    }

    return 0;
}