WAP whether a number is prime or not using function.

CODE ::

#include <stdio.h>
#include <math.h>
int check (int p);
int main ()
{
    int n;
    printf("Enter a number: ");
    scanf("%d",&n);
    if(check(n))
        printf("Prime\n");
    else
        printf("Not prime\n");
    return 0;
}
int check (int p)
{
    int i,f=0,limit;
    limit=sqrt(p);
    for(i=2;i<=limit;i++)
    {
        if(p%i==0)
        {
            f=1;
            break;
        }
    }
    if(f==0)
        return 1;

    return 0;
}