diff --git a/06/f.py b/06/f.py
index c075e3f..36641d9 100644
--- a/06/f.py
+++ b/06/f.py
@@ -6,19 +6,15 @@ input = open(os.path.dirname(__file__) +
 index, index1, index2 = 0, 0, 0
 buf = []
 
-for x in input:
-    index += 1
-    buf.append(x)
-    if index1 == 0 and len(buf) == 4:
-        if len(set(buf)) == 4:
-            index1 = index
-        buf.pop(0)
 
-    if index2 == 0 and len(buf) == 14:
-        if len(set(buf)) == 14:
-            index2 = index
-            break
-        buf.pop(0)
+def findUniq(length):
+    ind = 0
+    for x in input:
+        ind += 1
+        if len(set(input[ind:ind+length])) == length:
+            return ind+length
+    return False
 
-print("First answer: " + str(index1))
-print("Second answer: " + str(index2))
+
+print("First answer: " + str(findUniq(4)))
+print("Second answer: " + str(findUniq(14)))