diff --git a/wani.py b/wani.py
index a622d3c..e9c7f06 100644
--- a/wani.py
+++ b/wani.py
@@ -2,12 +2,13 @@ from xml.etree import ElementTree
 import re
 from enum import Enum
 from collections import defaultdict
+import sys
 
 from msd_translate import MSD_TRANSLATE
 
 
-STRUKTURE = "Kolokacije_strukture_09_new-system.xml"
-STAVKI = "k2.xml"
+STAVKI = sys.argv[1]
+STRUKTURE = sys.argv[2] # "Kolokacije_strukture_09_new-system.xml"
 
 CODES = {
     "Noun": "N",