修改完善
Signed-off-by: weslie <yaoht13@mails.tsinghua.edu.cn>