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
    uint16_t max = 0,val = 0;
    for(int i = 0; i <= 15; i++){
        if(reg & (1 << i)){
            max = i;
        }
        if(reg == 0){
            reg &= ~(0xffff << 0);
            return reg;
        }

    }
    reg &= ~(0xffff << 0);
    reg |= (1 << max);
    return reg;
}

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

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