这怎么用调用函数判断一个数是否为素数，是则返回1，不是则返回0？

#include<stdio.h>
int prime(int a)
{int i;
for(i=2;i<a;i++)
if(a%i==0)break;
if(i>=a)
return 1;
else
return 0;
}
int main()
{
int n,m;
printf("input n:");
scanf("%d",&n);
m=prime(n);
if(m==1)printf("是素数");
if(m==0)printf("不是素数");
return 0;
}