neat solution

Signed-off-by: Wei Jin <wjin.cn@gmail.com>