diff --git a/src/llm_benchmark/control/single.py b/src/llm_benchmark/control/single.py
index 9a314e6..b290084 100644
--- a/src/llm_benchmark/control/single.py
+++ b/src/llm_benchmark/control/single.py
@@ -35,17 +35,13 @@ def max_list(v: List[int]) -> int:
 
     @staticmethod
     def sum_modulus(n: int, m: int) -> int:
-        """Sum of modulus of numbers from 0 to n
-
-        Args:
-            n (int): Number to sum up to
-            m (int): Modulus
-
-        Returns:
-            int: Sum of modulus of numbers from 0 to n
-        """
-        arr = []
-        for i in range(n):
-            if i % m == 0:
-                arr.append(i)
-        return sum(arr)
+            """Sum of modulus of numbers from 0 to n
+    
+            Args:
+                n (int): Number to sum up to
+                m (int): Modulus
+    
+            Returns:
+                int: Sum of modulus of numbers from 0 to n
+            """
+            return sum(i for i in range(n) if i % m == 0)
\ No newline at end of file