diff --git a/math_probability/generate_primes/check_prime_challenge.ipynb b/math_probability/generate_primes/check_prime_challenge.ipynb
index e2b748a..a9aaf97 100644
--- a/math_probability/generate_primes/check_prime_challenge.ipynb
+++ b/math_probability/generate_primes/check_prime_challenge.ipynb
@@ -50,7 +50,7 @@
     "\n",
     "* None -> Exception\n",
     "* Not an int -> Exception\n",
-    "* 20 -> 2, 3, 5, 7, 11, 13, 17, 19"
+    "* 20 -> [False, False, True, True, False, True, False, True, False, False, False, True, False, True, False, False, False, True, False, True]"
    ]
   },
   {
diff --git a/math_probability/generate_primes/check_prime_solution.ipynb b/math_probability/generate_primes/check_prime_solution.ipynb
index 6f7a58b..1dcc4c6 100644
--- a/math_probability/generate_primes/check_prime_solution.ipynb
+++ b/math_probability/generate_primes/check_prime_solution.ipynb
@@ -49,7 +49,7 @@
     "\n",
     "* None -> Exception\n",
     "* Not an int -> Exception\n",
-    "* 20 -> 2, 3, 5, 7, 11, 13, 17, 19"
+    "* 20 -> [False, False, True, True, False, True, False, True, False, False, False, True, False, True, False, False, False, True, False, True]"
    ]
   },
   {