Refactor: make callback functions take an additional `apply_fn` parameter.

[gerdm]

User should not have to pass the apply_fn to the callback. This should be done automatically.