2015 day 6 refactored a bit

2015/day-6/part-1.py

@@ -24,7 +24,4 @@ for line in lines:
                 lights[(i, j)] = False
             else:
                 lights[(i, j)] = not lights[(i, j)]
-count = 0
-for v in lights.values():
-    count += v
-print(count)
+print(sum(lights.values()))

2015/day-6/part-2.py

@@ -27,7 +27,5 @@ for line in lines:
                     lights[(i, j)] -= 1
             else:
                 lights[(i, j)] += 2
-count = 0
-for v in lights.values():
-    count += v
-print(count)
+
+print(sum(lights.values()))