diff --git a/maths/sum_of_digits.py b/maths/sum_of_digits.py
index 5ad5fe6c9..d5488bb9e 100644
--- a/maths/sum_of_digits.py
+++ b/maths/sum_of_digits.py
@@ -1,10 +1,6 @@
-from timeit import timeit
-
-
 def sum_of_digits(n: int) -> int:
     """
     Find the sum of digits of a number.
-
     >>> sum_of_digits(12345)
     15
     >>> sum_of_digits(123)
@@ -25,7 +21,6 @@ def sum_of_digits(n: int) -> int:
 def sum_of_digits_recursion(n: int) -> int:
     """
     Find the sum of digits of a number using recursion
-
     >>> sum_of_digits_recursion(12345)
     15
     >>> sum_of_digits_recursion(123)
@@ -42,7 +37,6 @@ def sum_of_digits_recursion(n: int) -> int:
 def sum_of_digits_compact(n: int) -> int:
     """
     Find the sum of digits of a number
-
     >>> sum_of_digits_compact(12345)
     15
     >>> sum_of_digits_compact(123)
@@ -57,93 +51,24 @@ def sum_of_digits_compact(n: int) -> int:
 
 def benchmark() -> None:
     """
-    Benchmark code for comparing 3 functions,
-    with 3 different length int values.
+    Benchmark multiple functions, with three different length int values.
     """
-    print("\nFor small_num = ", small_num, ":")
-    print(
-        "> sum_of_digits()",
-        "\t\tans =",
-        sum_of_digits(small_num),
-        "\ttime =",
-        timeit("z.sum_of_digits(z.small_num)", setup="import __main__ as z"),
-        "seconds",
-    )
-    print(
-        "> sum_of_digits_recursion()",
-        "\tans =",
-        sum_of_digits_recursion(small_num),
-        "\ttime =",
-        timeit("z.sum_of_digits_recursion(z.small_num)", setup="import __main__ as z"),
-        "seconds",
-    )
-    print(
-        "> sum_of_digits_compact()",
-        "\tans =",
-        sum_of_digits_compact(small_num),
-        "\ttime =",
-        timeit("z.sum_of_digits_compact(z.small_num)", setup="import __main__ as z"),
-        "seconds",
-    )
+    from collections.abc import Callable
+    from timeit import timeit
 
-    print("\nFor medium_num = ", medium_num, ":")
-    print(
-        "> sum_of_digits()",
-        "\t\tans =",
-        sum_of_digits(medium_num),
-        "\ttime =",
-        timeit("z.sum_of_digits(z.medium_num)", setup="import __main__ as z"),
-        "seconds",
-    )
-    print(
-        "> sum_of_digits_recursion()",
-        "\tans =",
-        sum_of_digits_recursion(medium_num),
-        "\ttime =",
-        timeit("z.sum_of_digits_recursion(z.medium_num)", setup="import __main__ as z"),
-        "seconds",
-    )
-    print(
-        "> sum_of_digits_compact()",
-        "\tans =",
-        sum_of_digits_compact(medium_num),
-        "\ttime =",
-        timeit("z.sum_of_digits_compact(z.medium_num)", setup="import __main__ as z"),
-        "seconds",
-    )
+    def benchmark_a_function(func: Callable, value: int) -> None:
+        call = f"{func.__name__}({value})"
+        timing = timeit(f"__main__.{call}", setup="import __main__")
+        print(f"{call:56} = {func(value)} -- {timing:.4f} seconds")
 
-    print("\nFor large_num = ", large_num, ":")
-    print(
-        "> sum_of_digits()",
-        "\t\tans =",
-        sum_of_digits(large_num),
-        "\ttime =",
-        timeit("z.sum_of_digits(z.large_num)", setup="import __main__ as z"),
-        "seconds",
-    )
-    print(
-        "> sum_of_digits_recursion()",
-        "\tans =",
-        sum_of_digits_recursion(large_num),
-        "\ttime =",
-        timeit("z.sum_of_digits_recursion(z.large_num)", setup="import __main__ as z"),
-        "seconds",
-    )
-    print(
-        "> sum_of_digits_compact()",
-        "\tans =",
-        sum_of_digits_compact(large_num),
-        "\ttime =",
-        timeit("z.sum_of_digits_compact(z.large_num)", setup="import __main__ as z"),
-        "seconds",
-    )
+    for value in (262144, 1125899906842624, 1267650600228229401496703205376):
+        for func in (sum_of_digits, sum_of_digits_recursion, sum_of_digits_compact):
+            benchmark_a_function(func, value)
+        print()
 
 
 if __name__ == "__main__":
-    small_num = 262144
-    medium_num = 1125899906842624
-    large_num = 1267650600228229401496703205376
-    benchmark()
     import doctest
 
     doctest.testmod()
+    benchmark()