tidy codes

Signed-off-by: Clark Wang <clark.wangs@gmail.com>