Set Baud Rate Field in Control Register

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

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

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