Adding the optional step on register