Resolve "add callback function in step method to modify state"

Summary

(Summarize the merge request concisely.)

Explain your context

(Explain what you did.)

Please list breaking changes, new features or bugfixes

• Breaking changes:

• New features:

• Bugfixes:

Closes #354