Added callback function to set step to done