Set Baud Rate Field in Control Register

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

//0000 1111 1111 reg
//0000 0000 1111 baud
//0000 1111 1111 inv mask
//0000 1111 1111 reg = reg & inv mask
//1111 0000 0000 shifted baud
//1111 1111 1111 reg = reg | baud

uint32_t set_baud_rate(uint32_t reg, uint8_t baud) {
    // Your code here
    uint32_t baud1 = baud;
    uint32_t mask = ~(((1<<4)-1)<<8);
    reg=reg&mask;
    baud1=baud1<<8;
    reg = reg|baud1;
    return reg;
}

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