Count Set Bits in an 8-bit Register

Code

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

uint8_t count_set_bits(uint8_t reg) {
   int i;
   int count =0;
   for(i=0;i<8;i++){
    int ex;
    ex = reg & 0x01<<i;
    if(ex) count++;
   }
   return count;
}

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

Solving Approach