1 files changed, 2 insertions, 5 deletions
diff --git a/20/py/d07.py b/20/py/d07.py
index 775b7c8..1d25e53 100644
--- a/20/py/d07.py
+++ b/20/py/d07.py
@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+import aoc20
 import sys
 import regex as re
 import functools
@@ -43,11 +44,7 @@ def pt2(_in):
 
 
 if __name__ == "__main__":
-    if len(sys.argv) > 1:
-        input = open(sys.argv[1], "r").readlines()
-    else:
-        input = open("../input/07", "r").readlines()
+    input = aoc20.read_input(sys.argv[1:], 7)
     # graph(parse(input))
-
     print(pt1(input))
     print(pt2(input))