Extract the Nibble from an 8-bit Register

Code

#include <stdio.h>

unsigned char extractNibble(unsigned char reg, int pos) {
    // Write your code here
    unsigned char modified_reg = reg;
    if (pos) {
        modified_reg = modified_reg >> 4;
    } else {
        modified_reg = modified_reg & 0x0F;
    }

    return modified_reg;
}

int main() {
    unsigned char reg;
    int pos;
    scanf("%hhu %d", &reg, &pos);
    printf("%d", extractNibble(reg, pos));
    return 0;
}

Solving Approach