diff --git a/du5/a_station.c b/du5/a_station.c
index a7af519..1b4da0c 100644
--- a/du5/a_station.c
+++ b/du5/a_station.c
@@ -109,13 +109,9 @@ int count_targets(struct station* station){
 }
 
 int count_capacity(struct station* station){
-    int res  = 0;
-    struct car* ptr = *station->tracks;
-    while (ptr){
-        res += ptr->capacity;
-        ptr = ptr->next;
-
-    }
-    return res;
+    int total = 0;
+for (struct car* ptr = *station->tracks; ptr; ptr = ptr->next)
+    total += ptr->capacity;
+return total;
 }