diff --git a/README.md b/README.md
index 0177659..0e3864b 100644
--- a/README.md
+++ b/README.md
@@ -20,6 +20,10 @@ GPR-SIDL-inv/
 
 ├── dataset/                   # Training/testing data and synthetic datasets
 
+    ├── data.csv                   # Already generated dataset (70MHz Ricker wavelet)
+
+    ├── label.csv                   # Already generated dataset (70MHz Ricker wavelet) 
+
 ├── field_data/                # Used to store field data for inversion
 
 ├── gprMax/                    # Forward modeling package developed by the University of Edinburgh
@@ -40,6 +44,9 @@ GPR-SIDL-inv/
 
 ├── utils/                     # tool kit
 
+    ├── Model.py/                     # network model with transformer
+    
+    ├── Mydataset.py/                     # Loading and preprocessing datasets
 
 ├── 1_model_generator.py           # Randomly generate in files as needed to support forward modeling of gprMax in 3D media.
 
@@ -84,6 +91,8 @@ pip install -r requirements.txt
 
 ### 📊 Usage
 
+We provide an interface program for calling gprMax to generate custom datasets. If you need to generate a dataset, you can start running it from the first step. If you already have a simulated dataset, you can skip the first three steps. Steps four to six are used for personalized processing of our measured data, and we provide a frozen soil area measured data for testing. If you need to test new field data, please make targeted modifications.
+
 ### 1. Dataset generation
 
 .......