diff --git a/python/e087.py b/python/e087.py
index 61ddca9..bccbcc1 100644
--- a/python/e087.py
+++ b/python/e087.py
@@ -27,4 +27,4 @@ if __name__ == "__main__":
     solution = euler_087()
     print("e087.py: " + str(solution))
     assert(solution == 1097343)
-# 
+
diff --git a/python/lib_prime.py b/python/lib_prime.py
index 66db518..1cb33db 100644
--- a/python/lib_prime.py
+++ b/python/lib_prime.py
@@ -117,6 +117,24 @@ def primes(n_max):
     return ps
 
 
+def gen_primes():
+    """
+    Prime number generator function>
+    """
+    primes = [2]
+    yield 2
+    p = 3
+    while True:
+        for i in primes:
+            if p % i == 0:
+                break
+            if i * i > p:
+                primes.append(p)
+                yield p
+                break
+        p += 2
+
+
 def get_divisors_count(n):
     """
     Returns the number of divisors for n.