diff --git a/Loops/prime.c b/Loops/prime.c
index 2533f73..15f34cc 100644
--- a/Loops/prime.c
+++ b/Loops/prime.c
@@ -1,19 +1,48 @@
-//Find if no is prime or not
+
 #include<stdio.h>
 int main(){
-    int f=0,i,num;
-    printf("Enter number\n");
+    int f=0,i,num,flash=0;
+    printf("Enter a whole number\n");
     scanf("%d",&num);
-    for(i=2;i<=num/2;i++){
-        if(num%i==0){
-            f=1;
+    if(num==0)
+    {
+       printf("Zero is neither prime nor composite\n");
+    }
+   else if(num==1)
+    {
+       printf("One is neither prime nor composite\n");
+    }
+   else if(num==2)
+    {
+        printf("The entered number is a prime number ");
+    }
+    else if(num==3)
+    {
+        printf("The entered number is a prime number");
+    }
+    else
+    {
+        for(i=2;i<=num/2;i++){
+        if((num%i)==0)
+        {
+            flash=0;
             break;
         }
+        else
+        {
+            flash=1;
+        }
+
+
+
+    }
+    if(flash==0)
+    {
+        printf("The entered number is composite");
     }
-    if(f==0){
-        printf("Prime no");
+    else
+    {
+        printf("The entered number is prime");
     }
-    else{
-        printf("Not prime");
     }
-}
\ No newline at end of file
+}