diff --git a/DIRECTORY.md b/DIRECTORY.md
index 76c7f9dea4e3..5f314c31745d 100644
--- a/DIRECTORY.md
+++ b/DIRECTORY.md
@@ -995,6 +995,7 @@
     * [Sol1](project_euler/problem_686/sol1.py)
 
 ## Quantum
+  * [Bb84](quantum/bb84.py)
   * [Deutsch Jozsa](quantum/deutsch_jozsa.py)
   * [Half Adder](quantum/half_adder.py)
   * [Not Gate](quantum/not_gate.py)
diff --git a/maths/prime_numbers.py b/maths/prime_numbers.py
index 4e076fe317b4..c5297ed9264c 100644
--- a/maths/prime_numbers.py
+++ b/maths/prime_numbers.py
@@ -90,32 +90,20 @@ def fast_primes(max_n: int) -> Generator[int, None, None]:
             yield i
 
 
+def benchmark():
+    """
+    Let's benchmark our functions side-by-side...
+    """
+    from timeit import timeit
+
+    setup = "from __main__ import slow_primes, primes, fast_primes"
+    print(timeit("slow_primes(1_000_000_000_000)", setup=setup, number=1_000_000))
+    print(timeit("primes(1_000_000_000_000)", setup=setup, number=1_000_000))
+    print(timeit("fast_primes(1_000_000_000_000)", setup=setup, number=1_000_000))
+
+
 if __name__ == "__main__":
     number = int(input("Calculate primes up to:\n>> ").strip())
     for ret in primes(number):
         print(ret)
-
-    # Let's benchmark them side-by-side...
-    from timeit import timeit
-
-    print(
-        timeit(
-            "slow_primes(1_000_000_000_000)",
-            setup="from __main__ import slow_primes",
-            number=1_000_000,
-        )
-    )
-    print(
-        timeit(
-            "primes(1_000_000_000_000)",
-            setup="from __main__ import primes",
-            number=1_000_000,
-        )
-    )
-    print(
-        timeit(
-            "fast_primes(1_000_000_000_000)",
-            setup="from __main__ import fast_primes",
-            number=1_000_000,
-        )
-    )
+    benchmark()