update the scratch code

Signed-off-by: Levi Wang <wangli5665@gmail.com>