diff --git a/unit01_cipher_fundamentals/src/d_01.py b/unit01_cipher_fundamentals/src/d_01.py
index 59b68eb..b226615 100644
--- a/unit01_cipher_fundamentals/src/d_01.py
+++ b/unit01_cipher_fundamentals/src/d_01.py
@@ -12,15 +12,10 @@ def get_if_prime(val):
 
 	for k in range(0, 10000):
 		testval = 6 * k + 1;
-                if (testval>max):
-			break
-                if (val % testval == 0):
+        if (val % testval == 0):
 			return (False)
-                testval = 6 * k - 1;
-                if (testval>max):
-			break
- 
-                if (val % testval == 0):
+        testval = 6 * k - 1;
+        if (val % testval == 0):
 			return (False)
 	return (true)