diff --git a/Prime_number.py b/Prime_number.py
index 92800c63e83..1d9b03d79d9 100644
--- a/Prime_number.py
+++ b/Prime_number.py
@@ -19,15 +19,14 @@ def is_prime_a(n):
 
 
 def is_prime_b(n):
-    if n > 1:
-        if n == 2:
-            return True
-        else:
-            for i in range(2, int(n//2)+1):
-                if n % i == 0:
-                    return False
-            return True
-    return False
+    if n <= 1:
+        return False
+    if n == 2:
+        return True
+    for i in range(2, int(n//2)+1):
+        if n % i == 0:
+            return False
+    return True
 
 
 def is_prime_c(n):