diff --git a/README.md b/README.md
index 5ce765b4f1c0c4d8e845d1ebff8cc0cb7a4002ab..443286f8c82dd787dee62dafde4c8d29a14e26ff 100644
--- a/README.md
+++ b/README.md
@@ -7,17 +7,18 @@ pip install -r requirements.txt
 ```
 
 ### Data
-Dataset here is Cylinder2D.
+Dataset here is Cylinder2D with 3 channels.
+I will finish a script to download the dataset I used in the future.
 
-### Traning
+### Training
 
 All .py are in the folder DDPM.
 
-Use ddpm_conditional.py to train DDPM with conditional information through CFG.
+Use DDPM/ddpm_conditional.py to train DDPM with conditional information through CFG.
 
-Use interpolate.py to train Interpolator.
+Use DDPM/interpolate.py to train Interpolator.
 
-Use forecast.py to train Forecaster.
+Use DDPM/forecast.py to train Forecaster.
 
 ### Result