diff --git a/Loops/prime.c b/Loops/prime.c
index 2533f73..342d3ce 100644
--- a/Loops/prime.c
+++ b/Loops/prime.c
@@ -1,19 +1,37 @@
-//Find if no is prime or not
 #include<stdio.h>
 int main(){
     int f=0,i,num;
-    printf("Enter number\n");
+    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)
+        {
+            printf("The entered number is a composite number");
+            break;
+        }
+        else
+        {
+            printf("The entered number is a prime number");
             break;
         }
     }
-    if(f==0){
-        printf("Prime no");
-    }
-    else{
-        printf("Not prime");
     }
-}
\ No newline at end of file
+}