Add more improvement.

Signed-off-by: cctsao1008 <chiacheng.tsao@gmail.com>