Add callback function for testing purposes