diff --git a/06/tuning.py b/06/tuning.py
index 8a5f7ce..f68a1c7 100644
--- a/06/tuning.py
+++ b/06/tuning.py
@@ -2,18 +2,13 @@
 import fileinput
 from typing import Iterable
 
-def solve(data: str) -> int:
+def solve(data: str, length: int = 4) -> int:
     for idx in range(len(data)):
-        if len(set(list(data[idx:idx+4]))) == 4:
-            return idx + 4
-
-def solveb(data: str) -> int:
-    for idx in range(len(data)):
-        if len(set(list(data[idx:idx+14]))) == 14:
-            return idx + 14
+        if len(set(list(data[idx:idx+length]))) == length:
+            return idx + length
 
 if __name__ == "__main__":
     with fileinput.input() as f:
         data = f.readline()
-    print(solve(data))
-    print(solveb(data))
+    print(solve(data, 4))
+    print(solve(data, 14))