function that returns the k(th) digit(from the right) in n(a positive integer)

#include <stdio.h>

int digit(int n, int k)
{
  
    while(k>1){
        n/=10;
        k--;
    }
    
    return n%10;
}

int main(){
    int n, k;
    
    printf("Enter n, k: ");
    scanf("%d %d", &n, &k);
    
    printf("%d", digit(n,k));
    return 0;
}