Convert Decimal Number to Binary or Hex Without itoa function

#include <stdio.h>
#include <stdint.h>

void print_base(uint16_t num, uint8_t base) {
    // Your logic here
             char hex[32];
    int i = 0;
        if(num == 0){
        printf("0\n");
    }
    if(base == 2){
        while(num > 0){
            hex[i++] = (num & 1);
           num >>=1; 
        }
            for (int j = i - 1; j >= 0; j--) {
        printf("%d", hex[j]);
    }
    } else {

    while (num > 0) {
        int rem = num % 16;
        hex[i++] = (rem < 10) ? (rem + '0') : (rem + 'A' - 10);
        num /= 16;
    }
    for (int j = i - 1; j >= 0; j--) {
        printf("%c", hex[j]);
    }
    printf("\n");

    }
}

int main() {
    uint16_t num;
    uint8_t base;
    scanf("%hu %hhu", &num, &base);

    print_base(num, base);
    return 0;
}