C语言函数设计判断素数

#include <stdio.h>
#include <math.h>
int IsPrimeNumber(int number);
int main(void)
{
	int n, ret;
	printf("Input a positive integer,a:");
	scanf("%d", &n);
	if (IsPrimeNumber(n))
    {
        printf("\n%d is a prime number",n);
    }
    else
    {
        printf("\n%d is not a prime number",n);
    }
     return 0;
}

int IsPrimeNumber(int number)
{
    int flag=1;
    int i;
    if(number<2)
    {
        flag=0;
    }
    for(i=2;i<number;i++)
    {
        if(number%i==0)
        {
            flag=0;
            break;
        }
    }
    return flag;
}