diff --git a/loops/py/code.py b/loops/py/code.py
index 119c0060..c707b4da 100644
--- a/loops/py/code.py
+++ b/loops/py/code.py
@@ -1,15 +1,11 @@
-import sys
 import random
+import sys
 
 
 def main():
     u = int(sys.argv[1])  # Get an input number from the command line
     r = random.randint(0, 10000)  # Get a random number 0 <= r < 10k
-    a = [0] * 10000  # Array of 10k elements initialized to 0
-    for i in range(10000):  # 10k outer loop iterations
-        for j in range(100000):  # 100k inner loop iterations, per outer loop iteration
-            a[i] += j % u  # Simple sum
-        a[i] += r  # Add a random value to each element in array
+    a = [sum([j % u for j in range(100_000)]) + r for _ in range(10_000)]
     print(a[r])  # Print out a single element from the array