Keep Only the Highest Set Bit

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

// Complete the function
uint16_t highest_set_bit(uint16_t reg) {
    
    int i = 0;
    while (1) {
        if (reg == 0x01) {
            reg = reg << i;
            break;
        } else if (reg == 0x00) {
            break;
        } else {
            reg &= ~(0x01);
            reg = reg >> 1;
            i++;
        }
    }
    return reg;
}

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

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