diff --git a/day-02/day-02.py b/day-02/day-02.py
index 8ae9235..04a4711 100644
--- a/day-02/day-02.py
+++ b/day-02/day-02.py
@@ -4,13 +4,13 @@ def part_1(input):
     h_pos = 0
     depth = 0
     for line in input:
-        step = line.split()
-        if 'forward' == step[0]:
-            h_pos += int(step[1])
-        elif 'up' == step[0]:
-            depth -= int(step[1])
-        elif 'down' == step[0]:
-            depth += int(step[1])
+        cmd, rng = line.split()
+        if 'forward' == cmd:
+            h_pos += int(rng)
+        elif 'up' == cmd:
+            depth -= int(rng)
+        elif 'down' == cmd:
+            depth += int(rng)
     result = h_pos * depth
     print("Part 1 result:", result)
 
@@ -20,14 +20,14 @@ def part_2(input):
     depth = 0
     aim = 0
     for line in input:
-        step = line.split()
-        if 'forward' == step[0]:
-            h_pos += int(step[1])
-            depth += aim * int(step[1])
-        elif 'up' == step[0]:
-            aim -= int(step[1])
-        elif 'down' == step[0]:
-            aim += int(step[1])
+        cmd, rng = line.split()
+        if 'forward' == cmd:
+            h_pos += int(rng)
+            depth += aim * int(rng)
+        elif 'up' == cmd:
+            aim -= int(rng)
+        elif 'down' == cmd:
+            aim += int(rng)
     result = h_pos * depth
     print("Part 2 result:", result)