diff --git a/cv6/main.c b/cv6/main.c
index 4ede202..63fe14f 100644
--- a/cv6/main.c
+++ b/cv6/main.c
@@ -5,6 +5,8 @@
 #include <string.h>
 #include "a_station.h"
 
+#define MAX 1000
+
 void print_station(struct station* station){
 	// Vypise celu stanicu
 	printf("station>>>\n");
@@ -23,42 +25,60 @@ void print_station(struct station* station){
 }
 
 void test_station(struct station* station){
-   while(1){
-       char car[30];
-       fgets(car, 30, stdin);
-       car[strlen(car)-1] = '\0';
-       if(car[0] == '\0'){
-           break;
-       }
-       char pass[30];
-       if(!isdigit(car[0])){
-           fgets(pass, 30, stdin);
-           pass[strlen(car)-1] = '\0';
-           int num = atoi(pass);
-           add_target_capacity(station, car, num);
-       }
-
-   }
+    FILE *fp = fopen("station.txt", "r");
+    if(fp == NULL){
+        printf("Nemozem najst stanice\n");
+        exit(1);
+    }  
+    int i = 0;
+    while(1){
+        
+        char s[30];
+        char *r = fgets(s, 30, fp);
+        s[strlen(s)-1] = '\0';
+         //printf("%s", car);
+        
+        if(r == NULL){
+            if(feof(fp) != 0){
+                printf("End of file\n");
+                break;
+            } 
+            else{
+                printf("Error\n"); 
+                exit(0);
+            }
+        }
+        char pass[10];
+        if(!isdigit(s[i])){
+            fgets(pass, 10, fp);
+            pass[strlen(pass)-1] = '\0';
+            int num = atoi(pass);
+            add_target_capacity(station, s, num); 
+        } 
    
+    }
+
 }
 
 int main(){
       
-    printf("Zadajte zoznam cieľových staníc a počet cestujúcich.\n");
-    printf("Zoznam zakončite prázdnym riadkom.\n");
     struct station* station = create_station();   
     test_station(station);
     
     printf("Zadajte hľadanu stanicu:\n");
-    char clear[30];
-    fgets(clear,30,stdin);
-    clear[strlen(clear)-1] = '\0';  
+    char find[30];
+    fgets(find,30,stdin);
+    find[strlen(find)-1] = '\0';  
     
     printf("Výsledný vlak bude:\n"); 
-    int s = get_target_capacity(station, clear);
+    int s = get_target_capacity(station, find);
     print_station(station);
-    printf("Capacita na danej statice: %d\n", s); 
-    
+    if(s == 0){
+        printf("Neexistuje taka stanica\n");
+    }
+    else{
+        printf("Capacita na danej statice: %d\n", s); 
+    }
     destroy_station(station);
     return 0;
 }