Add a step to define a callback function