prime_or_not

#include<stdio.h>

int main(){
    int i,n,prime=1;
    printf("Enter number");
    scanf("%d",&n);

    if(n==1){
        printf("1 is not a prime number");
    }

    else if(n==2){
        printf("2 is not a prime number");
    }

    else{
        for (i=2; i<n; i++)
        {
            if(n%i==0){
                prime=0;
                break;
            }
        }

        if(prime==1){
            printf("%d is prime\n", n);
        }
        else{
            printf("%d is not prime \n", n);
        }  
    }
    return 0;
}