Add test for callback
Add tests