diff --git a/data.sh b/data.sh
index 4aa0ed1..4b53e8c 100644
--- a/data.sh
+++ b/data.sh
@@ -8,4 +8,7 @@ wc -l ./vaccines.csv
 head -n 1468 vaccines.csv > ./test.csv
 head -n 2936 vaccines.csv | tail -n 1468 > ./validation.csv
 tail -n +2936 vaccines.csv > ./train.csv
-wc -l ./*.csv
\ No newline at end of file
+wc -l ./*.csv
+sed -i '1s/^/country,iso_code,date,total_vaccinations,people_vaccinated,people_fully_vaccinated,daily_vaccinations_raw,daily_vaccinations,total_vaccinations_per_hundred,people_vaccinated_per_hundred,people_fully_vaccinated_per_hundred,daily_vaccinations_per_million,vaccines,source_name,source_website\n/' ./test.csv
+sed -i '1s/^/country,iso_code,date,total_vaccinations,people_vaccinated,people_fully_vaccinated,daily_vaccinations_raw,daily_vaccinations,total_vaccinations_per_hundred,people_vaccinated_per_hundred,people_fully_vaccinated_per_hundred,daily_vaccinations_per_million,vaccines,source_name,source_website\n/' ./validation.csv
+sed -i '1s/^/country,iso_code,date,total_vaccinations,people_vaccinated,people_fully_vaccinated,daily_vaccinations_raw,daily_vaccinations,total_vaccinations_per_hundred,people_vaccinated_per_hundred,people_fully_vaccinated_per_hundred,daily_vaccinations_per_million,vaccines,source_name,source_website\n/' ./train.csv
\ No newline at end of file