diff --git a/script2.sh b/script2.sh
index 86c300d..82fcbf7 100644
--- a/script2.sh
+++ b/script2.sh
@@ -1,4 +1,6 @@
 #!/bin/sh
 
 
-wc -l > num_of_lines.txt
\ No newline at end of file
+wc -l TRAIN_winequality-red.csv > num_of_lines.txt
+wc -l TEST_winequality-red.csv >> num_of_lines.txt
+wc -l VAL_winequality-red.csv >> num_of_lines.txt