diff --git a/AoC2021/template.py b/AoC2021/template.py
index 4ce60d8..89cb6ed 100644
--- a/AoC2021/template.py
+++ b/AoC2021/template.py
@@ -4,8 +4,7 @@ import sys
 
 
 if __name__ == '__main__':
-    lines = open(sys.argv[1]).read().strip('\n\n').split('\n') # or split \n\n or sth similar
-    nums = [int(i) for i in lines]
+    nums = [int(line.strip('\n')) for line in open(sys.argv[1])]
 
     # challenge 1
     res1 = ""