Write a function that returns 1 if the number is prime and 0 if not prime. Number is passed to the function as argument.

#include <stdio.h>
#include <math.h>
int prime(int n);
int main ()
{
    int n,res;
    scanf("%d",&n);
    res=prime(n);
    printf("%d",res);
    return 0;
}
int prime(int n)
{
    int flag=1,m,i;
    m=sqrt(n);
    for(i=2;i<=m;i++)
    {
        if(n%i==0)
        {
            flag=0;
            break;
        }
    }
    if(flag==1)
        return 1;
    else
        return 0;
}