What's the use case for the callback you've added? Not sure to understand the use for it...