Count Set Bits in an 8-bit Register

Code

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

uint8_t count_set_bits(uint8_t reg) {
    uint8_t count = 0; 
    uint8_t mask = 0x1; 
    for (int i = 0; i < 8; i++) { 
        if (reg & mask){ 
            count ++;
        }
        reg = (reg >> 1); 
    }
    return count;
}

int main() {
    uint8_t reg;
    scanf("%hhu", &reg);
    printf("%u", count_set_bits(reg));
    return 0;
}

Solving Approach