2 files changed, 9 insertions, 2 deletions

diff --git a/aoc/__main__.py b/aoc/__main__.py
index befa8ec..25a7026 100644
--- a/aoc/__main__.py
+++ b/aoc/__main__.py
@@ -1,6 +1,7 @@
 import argparse
 import importlib
 import os
+from time import perf_counter
 from typing import List
 
 
@@ -34,8 +35,13 @@ if __name__ == '__main__':
         **dict(args.kwargs)
     )
 
+    start = perf_counter()
     print(
         assignment.run(
             input=assignment.read_input(example=args.example),
         )
     )
+    end = perf_counter()
+    delta = end - start
+    print()
+    print(f'{(delta if delta > 1 else delta * 1000):.3f}{"s" if delta > 1 else "ms"}')
diff --git a/day7/__init__.py b/day7/__init__.py
index b3579b7..933b9e0 100644
--- a/day7/__init__.py
+++ b/day7/__init__.py
@@ -37,5 +37,6 @@ class AssignmentTwo(Assignment):
     @staticmethod
     def calculate_cost(start, end) -> int:
         base_cost = AssignmentOne.calculate_cost(start, end)
-        extra = sum(i for i in range(base_cost))
-        return base_cost + extra
\ No newline at end of file
+        # extra = sum(i for i in range(base_cost))
+        # return base_cost + extra
+        return int(base_cost * (base_cost + 1) / 2)