diff --git a/solution.py b/solution.py
index cb3908c..e9056ce 100644
--- a/solution.py
+++ b/solution.py
@@ -2,10 +2,7 @@
 import sys
 from pprint import pprint
 
-input_f = "input" #sys.argv[1]
-
-with open(input_f) as file:
-    for line in file:
+input_f = sys.argv[1]
 
 
 #########################################
@@ -14,6 +11,9 @@ with open(input_f) as file:
 #                                       #
 #########################################
 
+with open(input_f) as file:
+    for line in file:
+