diff --git a/2.py b/2.py
index af4ffa8..1e141e0 100644
--- a/2.py
+++ b/2.py
@@ -1,6 +1,2 @@
 with open("input2_pre.txt") as f:
-    s = 0
-    for line in f.readlines():
-        a, b = [int(a) for a in line.strip().split()]
-        s += b + ((3, 0, 6)[((a - b) % 3)])
-    print(s)
\ No newline at end of file
+    print(sum([(([int(a) for a in line.strip().split()][1]) + ((3, 0, 6)[((([int(a) for a in line.strip().split()][0]) - ([int(a) for a in line.strip().split()][1])) % 3)])) for line in f.readlines()]))
\ No newline at end of file
diff --git a/2pt2.py b/2pt2.py
index d5a8c19..73427d8 100644
--- a/2pt2.py
+++ b/2pt2.py
@@ -1,17 +1,2 @@
 with open("input2_pre.txt") as f:
-    s = 0
-    for line in f.readlines():
-        a, b = [int(a) for a in line.strip().split()]
-        b = {
-            (1, 1): 3,
-            (1, 2): 1,
-            (1, 3): 2,
-            (2, 1): 1,
-            (2, 2): 2,
-            (2, 3): 3,
-            (3, 1): 2,
-            (3, 2): 3,
-            (3, 3): 1,
-        }[(a, b)]
-        s += b + ((3, 0, 6)[((a - b) % 3)])
-    print(s)
\ No newline at end of file
+    print(sum([(({(1, 1): 3,(1, 2): 1,(1, 3): 2,(2, 1): 1,(2, 2): 2,(2, 3): 3,(3, 1): 2,(3, 2): 3,(3, 3): 1,}[([int(a) for a in line.strip().split()][0], [int(a) for a in line.strip().split()][1])]) + ((3, 0, 6)[((([int(a) for a in line.strip().split()][0]) - ({(1, 1): 3,(1, 2): 1,(1, 3): 2,(2, 1): 1,(2, 2): 2,(2, 3): 3,(3, 1): 2,(3, 2): 3,(3, 3): 1,}[([int(a) for a in line.strip().split()][0], [int(a) for a in line.strip().split()][1])])) % 3)])) for line in f.readlines()]))
\ No newline at end of file