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
    uint8_t bit = 0 ;
    for(uint8_t  i = 0 ; i < 4; i++)
    {
        bit = (baud >> i) & 1;
        if( bit == 1)
        {
            reg = reg | 1 << (8 + i);
        }
        else 
        {
            reg = reg & ~(1 << (8+i));
        }
    }
    return reg;
}

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