diff --git a/README.md b/README.md
index 26902d2..626b781 100644
--- a/README.md
+++ b/README.md
@@ -41,6 +41,10 @@ GPR-SIDL-inv/
 
     ├── Network/                       # Used for storing network models and data loading programs
 
+        ├── Model.py/                  # Network Model Based on Transformer and U-shaped Network Structure
+    
+        ├── Mydataset.py/              # Data loading and preprocessing
+
     ├── readgssi/                      # Software package for reading and converting raw data
 
     ├── SAVE/                          # Save the trained model
@@ -49,9 +53,9 @@ GPR-SIDL-inv/
 
     ├── utils/                         # tool kit
 
-        ├── Model.py/                  # network model with transformer
+        ├── layers_generator.py/       # Generate random layers and assign attribute programs
     
-        ├── Mydataset.py/              # Loading and preprocessing datasets
+        ├── train_val_lr.py/           # Network Training and Validation Management
 
         ├── plot.py/                   # the tool kit for plotting the 2D image