diff --git a/doc/design/cluster_train/pserver_client.md b/doc/design/cluster_train/pserver_client.md
index 56469fc21535ea1104afd9d02259484774d7b030..ee40eb32c55a41fa72ca1a1cc180421c0719c15f 100644
--- a/doc/design/cluster_train/pserver_client.md
+++ b/doc/design/cluster_train/pserver_client.md
@@ -6,10 +6,14 @@ For an overview of trainer's role, please refer to [distributed training design
 
 The parameters on parameter servers need to be initialized. To provide maximum flexibility, we need to allow trainer initialized the parameters. Only one trainer will do the initialization, the other trainers will wait for the completion of initialization and get the parameters from the parameter servers.
 
+### Trainer Selection
+
 To select the trainer for initialization, every trainer will try to get a distributed lock, whoever owns the lock will do the initialization. As illustrated below:
 
 <img src="./src/init_lock.png">
 
+### Selection Process
+
 The select process is encapsulated in the C API function:
 ```c
 int paddle_begin_init_params(paddle_pserver_client* client, const char* config_proto);