diff --git a/flask_company_data.py b/flask_company_data.py
index dd042c4..1e398a7 100644
--- a/flask_company_data.py
+++ b/flask_company_data.py
@@ -150,9 +150,9 @@ def check_polish_wl(nip: str):
     }
 
 
-@app.get("/")
-def index():
-    return render_template("company_data.html")
+@app.get("/company_data/<country>")
+def index(country):
+    return render_template("company_data.html", country=country)
 
 
 @app.get("/api/company_lookup")
diff --git a/templates/company_data.html b/templates/company_data.html
index d40dab9..fb4b8fb 100644
--- a/templates/company_data.html
+++ b/templates/company_data.html
@@ -14,9 +14,9 @@
         <div class="form-grid">
             <div class="row">
                 <select id="country">
-                    <option value="BE">Belgium</option>
-                    <option value="HR">Croatia</option>
-                    <option value="PL">Poland</option>
+                    <option value="BE" {% if country == "BE" %}selected{% endif %}>Belgium</option>
+                    <option value="HR" {% if country == "HR" %}selected{% endif %}>Croatia</option>
+                    <option value="PL" {% if country == "PL" %}selected{% endif %}>Poland</option>
                 </select>
                 <input id="vat_number" type="text" placeholder="VAT number">
                 <button type="button" onclick="searchCompany()">Search</button>