Moved the callback to be a general callback

Maybe in future if needed I’ll add a per row callback