Add callback mechanism for callbacks after steps / stages