Extract and Modify Field in a 32-bit Register

#include <stdio.h>
#include <stdint.h>
#define mask 0b11111 << 10

uint32_t update_register(uint32_t reg) {
    // Your logic here
    uint32_t layout = (reg & mask) >> 10;
    if (layout < 31){
        reg = reg & ~(mask);
        layout++;
        reg |= layout << 10;
    }

    return reg;
}

int main() {
    uint32_t reg;
    scanf("%u", &reg);
    uint32_t updated = update_register(reg);
    printf("%u", updated);
    return 0;
}