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
    int i = 16;
    int count = 0;
    while(i >= 0){
        //Detects First Set Bit
        if((reg & (1 << i)) && (count == 0)){
            count++;
            i--;
        //Clears Set Bits After First
        } else if((reg & (1 << i)) && (count > 0)) {
            reg &= ~(1 << i);
            i--;
        //Passes Over Unset Bits
        } else {
            i--;
        }
    }
    return reg;
}

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

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