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 << 4) - 1;        // Step 1
    reg &= ~(mask << 8);                  // Step 2 & 3 (clear bits)
    reg |= (baud & mask) << 8;           // Step 4 & 5 (set new value)

    return reg;
}

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