added final stage behaviour