diff --git a/cv4/a_train.c b/cv4/a_train.c
index e6c1958..1dd34aa 100644
--- a/cv4/a_train.c
+++ b/cv4/a_train.c
@@ -16,12 +16,12 @@ struct car* add_car(struct car* first, const char* target){
 
     struct car* temp = first;
 
-    while(temp != NULL)
+    while(temp->next != NULL)
         temp = temp->next;
 
-    temp = (struct car*)calloc(1, sizeof(struct car));
-    strcpy(temp->value, target);
-    temp->next = NULL;
+    temp->next = (struct car*)calloc(1, sizeof(struct car));
+    strcpy(temp->next->value, target);
+    temp->next->next = NULL;
     return first;
 }
 
@@ -73,13 +73,4 @@ struct car* clear_train(struct car* first, const char* target){
     }
 
     return first;
-}
-
-/*int main(){
-    struct car* myCar = (struct car*)calloc(1, sizeof(struct car));
-    myCar = add_car(myCar, "Kosice");
-    print_train(myCar);
-    myCar = clear_train(myCar, "Kosice");
-    print_train(myCar);
-    return 0;
-}*/
\ No newline at end of file
+}
\ No newline at end of file