Implement Custom strcmp function

#include <stdio.h>

int custom_strcmp(const char *a, const char *b) {
    // Your logic here
    int i = 0;
    
    // Compare characters until find a mismatch or reach end of either string
    while (a[i] != '\0' && b[i] != '\0') 
    {
        if (a[i] != b[i]) 
        {
            return a[i] - b[i];  // Return ASCII difference
        }
        i++;
    }
    
    // One or both strings ended, return difference of the characters at current position
    return a[i] - b[i];
}

int main() {
    char a[101], b[101];
    fgets(a, sizeof(a), stdin);
    fgets(b, sizeof(b), stdin);

    // Remove newline
    int i = 0;
    while (a[i]) {
        if (a[i] == '\n') { a[i] = '\0'; break; }
        i++;
    }
    i = 0;
    while (b[i]) {
        if (b[i] == '\n') { b[i] = '\0'; break; }
        i++;
    }

    printf("%d", custom_strcmp(a, b));
    return 0;
}