Print Binary Representation of an 8-bit or 16-bit Value

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

void print_binary(uint16_t val) {
    // Your logic here
    int size = 7;
    if(val > 255)
    {
        size = 15;
    }
    for(int i=size;i>=0;i--)
    {
        if((val >> i) & 0x01)
        {
            printf("1");
        }
        else 
        {
            printf("0");
        }
    }
}

int main() {
    uint16_t val;
    scanf("%hu", &val);
    print_binary(val);
    return 0;
}