added callback to functions in test