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){
        if(num == 0){
            printf("%d",num);
        }
        else if(num <= 15){
            for(int i=3;i>=0;i--){
                printf("%d",(num >> i)&1);
            }
        }else{
            for(int i=7;i>=0;i--){
                printf("%d",(num >> i)&1);
            }            
        }
    }else{
        printf("%X",num);
    }
}

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

    print_base(num, base);
    return 0;
}