From a35fe7321076a23dd15281d8452b5ee7f7842319 Mon Sep 17 00:00:00 2001
From: VittalKumar23 <109903004+VittalKumar23@users.noreply.github.com>
Date: Mon, 16 Oct 2023 11:32:49 +0530
Subject: [PATCH] Update P07_PrimeNumber.py

---
 Programs/P07_PrimeNumber.py | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/Programs/P07_PrimeNumber.py b/Programs/P07_PrimeNumber.py
index 485de60..766aa1d 100644
--- a/Programs/P07_PrimeNumber.py
+++ b/Programs/P07_PrimeNumber.py
@@ -1,23 +1,18 @@
-#Author: OMKAR PATHAK
-#This program checks whether the entered number is prime or not
-
 def checkPrime(number):
     '''This function checks for prime number'''
-    isPrime = False
-    if number == 2:
-        print(number, 'is a Prime Number')
+    isPrime = True  # Assume the number is prime by default
+
     if number > 1:
         for i in range(2, number):
             if number % i == 0:
-                print(number, 'is not a Prime Number')
                 isPrime = False
                 break
-            else:
-                isPrime = True
 
-        if isPrime:
-            print(number, 'is a Prime Number')
+    if isPrime:
+        print(number, 'is a Prime Number')
+    else:
+        print(number, 'is not a Prime Number')
 
-if __name__ == '__main__':
+if _name_ == '_main_':
     userInput = int(input('Enter a number to check: '))
     checkPrime(userInput)