自定义函数处理素数

写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息

#include<stdio.h>
void fun(int x)
{
int i,flag=1;
for(i=2;i<x;i++)
{
if(x%i)
flag=0;
break;
}
if(flag) 
printf("prime");
else printf("not prime");
}
int main()
{
int x;
scanf("%d",&x);
fun(x);
return 0;
}