17. Extract the Nibble from an 8-bit Register

#include <stdio.h>

unsigned char extractNibble(unsigned char reg, int pos) {
    // Write your code here
if(pos==1){
    unsigned char mask_1 = 0xf0;
    return (reg&mask_1)>>4;
}else{
    unsigned char mask_0 = 0x0f;
    return (reg&mask_0);
}
}

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