Update scritpt.

Signed-off-by: Ji You <jiyou09@gmail.com>
Update scritpt.

Signed-off-by: Ji You <jiyou09@gmail.com>