diff --git a/cv6/main.c b/cv6/main.c
index 948e527..c0ed024 100644
--- a/cv6/main.c
+++ b/cv6/main.c
@@ -1,5 +1,8 @@
+
 #include <stdio.h>
+#include <stdlib.h>
 #include "a_station.h"
+#include <string.h>
 
 int main() {
     // Vytvorenie stanice
@@ -9,22 +12,40 @@ int main() {
         return 1;
     }
 
-    // Pridanie nových záznamov
-    add_target_capacity(moja_stanica, "Bratislava", 50);
-    add_target_capacity(moja_stanica, "Košice", 30);
-    add_target_capacity(moja_stanica, "Prešov", 20);
+    // Načítanie údajov zo súboru
+    load_from_file(moja_stanica);
 
-    // Získanie informácií o mestách
-    printf("Počet cestujúcich do Bratislava: %d\n", get_target_capacity(moja_stanica, "Bratislava"));
-    printf("Počet cestujúcich do Košice: %d\n", get_target_capacity(moja_stanica, "Košice"));
-    printf("Počet cestujúcich do Prešov: %d\n", get_target_capacity(moja_stanica, "Prešov"));
+    // Cyklus pre zadávanie nových záznamov
+    char target[TARGET_SIZE];
+    int capacity;
+    while (1) {
+        printf("Zadajte názov stanice (alebo 'exit' pre ukončenie): ");
+        scanf("%s", target);
+        if (strcmp(target, "exit") == 0) {
+            break; // Ukončiť cyklus
+        }
 
-    // Výpočet celkového počtu cieľových staníc a cestujúcich
-    printf("Celkový počet cieľových staníc: %d\n", count_targets(moja_stanica));
-    printf("Celkový počet cestujúcich: %d\n", count_capacity(moja_stanica));
+        printf("Zadajte počet cestujúcich: ");
+        scanf("%d", &capacity);
+        add_target_capacity(moja_stanica, target, capacity);
+    }
+
+    // Výstup informácií o staniciach
+    printf("Uložené údaje:\n");
+    for (int i = 0; i < moja_stanica->track_count; i++) {
+        struct car* current = moja_stanica->tracks[i];
+        while (current != NULL) {
+            printf("Stanica: %s, Cestujúcich: %d\n", current->value, current->capacity);
+            current = current->next;
+        }
+    }
+
+    // Uloženie údajov do súboru pred ukončením
+    save_to_file(moja_stanica);
 
     // Uvoľnenie pamäte
     destroy_station(moja_stanica);
 
     return 0;
 }
+