Função Número Primo em C

Implementar uma função para testar se um número é primo.

#include<stdio.h>;

int primo(int n)
{
int i=2;
int z;

while(i<=n-1){
z = n%i;
if(z==0){
return 0;
}

i++;
}
return 1;
}

int main(){
int k;
printf("Digite um número");
scanf("%d",&k);
if(primo(k)==0){
printf("Não é primo.");
}else{
printf("É primo.");

}
}