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
    if(base == 2){
        int flag = 0;
        for(int i = 7;i >= 0; i--){
            int bit = (num >> i) & 1;
            if(bit == 1){
                flag = 1;
            }
            if(flag){
                printf("%d",bit);
            }
        }
        if(!flag){
            printf("0");
        }
    }
    else if(base == 16){
        printf("%X",num);
    }
}

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

    print_base(num, base);
    return 0;
}