Keep Only the Highest Set Bit

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

// Complete the function
uint16_t highest_set_bit(uint16_t reg) {
    // Your logic here
    uint8_t div_count = 0;
    if(reg == 0U){
        return reg;
    }
    while (reg >> 1U){
        reg = reg >> 1U;
        // printf("%hu", reg);
        ++div_count;
    }
    return (1U << div_count);
}

int main() {
    uint16_t reg;
    scanf("%hu", &reg);

    uint16_t result = highest_set_bit(reg);
    printf("%hu", result);
    return 0;
}