add callback

Signed-off-by: Kevin Luo <kevinluo201@gmail.com>