diff --git a/vcf_cleaner.py b/vcf_cleaner.py
index 0e8aea0..ae78a1d 100644
--- a/vcf_cleaner.py
+++ b/vcf_cleaner.py
@@ -1,33 +1,29 @@
+import os
+import sys
+import base64
 import pickle
 from pythonvCard4.vcard import Contact
-import base64
-import os
 
-file = open("contacts.vcf", 'r').readlines()
+input_file = open(sys.argv[1], 'r').readlines()
 
 cards = []
-
 current_card = ""
-for line in file:
+for line in input_file:
     current_card += line
     if "END:VCARD" in line:
         contact = Contact.from_vcard(current_card)
-        if "PHOTO" in contact.custom and len(contact.custom["PHOTO"]) > 1:
-            os.makedirs("photo/" + contact.fn, exist_ok=True)
-            for image in range(len(contact.custom["PHOTO"])):
-                with open("photo/" + contact.fn + "/" + str(image) + ".jpg", "wb") as f:
-                    f.write(base64.decodebytes(str.encode(contact.custom["PHOTO"][image])))
+        # if "PHOTO" in contact.custom and len(contact.custom["PHOTO"]) > 1:
+        #     os.makedirs("photo/" + contact.fn, exist_ok=True)
+        #     for image in range(len(contact.custom["PHOTO"])):
+        #         with open("photo/" + contact.fn + "/" + str(image) + ".jpg", "wb") as f:
+        #             f.write(base64.decodebytes(str.encode(contact.custom["PHOTO"][image])))
 
         cards.append(contact)
         current_card = ""
         continue
 
-
-# contact = Contact.from_vcard(cards[156])
-# print(contact.custom["PHOTO"])
-
-# vcf_text = contact.to_vcard()
-# print(vcf_text)
-# open('test.vcf', 'w').write(vcf_text)
-
-print(len(cards))
+output = open('output.vcf', 'w')
+for card in cards:
+    vcf_text = card.to_vcard()
+    print(vcf_text)
+    output.write(vcf_text)