Add test to check callback functionality.