add callback function in step method to modify state

Summary

While the call of the callback function is located in the reset function, the option is missing in the step function for the BaseEnvLive.