diff --git a/cv4/a_train.c b/cv4/a_train.c
index 3b66395..25ab7b9 100644
--- a/cv4/a_train.c
+++ b/cv4/a_train.c
@@ -7,7 +7,7 @@ struct car* add_car(struct car* first, const char* target){
     if(!strcmp(target, ""))
         return NULL;
 
-    if(!first){
+    if(!first || !strcmp(first->value, "")){
         first = (struct car*)calloc(1, sizeof(struct car));
         strcpy(first->value, target);
         first->next = NULL;
@@ -16,12 +16,12 @@ struct car* add_car(struct car* first, const char* target){
 
     struct car* temp = first;
 
-    while(temp->next != NULL)
+    while(temp != NULL)
         temp = temp->next;
 
-    temp->next = (struct car*)calloc(1, sizeof(struct car));
-    strcpy(temp->next->value, target);
-    temp->next->next = NULL;
+    temp = (struct car*)calloc(1, sizeof(struct car));
+    strcpy(temp->value, target);
+    temp->next = NULL;
     return first;
 }
 
@@ -59,12 +59,27 @@ struct car* clear_train(struct car* first, const char* target){
 
     while(temp != NULL){
         next = temp->next;
-        if(!strcmp(temp->value, target))
-        free(temp); 
+        if(!strcmp(temp->value, target)) {
+            if(temp == first) {
+                free(first);
+                first = NULL;
+            }
+            else
+                free(temp);
+        }
         temp = next;
         if(!first)
             first = temp;
     }
 
     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