Update code.

Signed-off-by: Wiki Wang <wikiwang@live.com>