Add workflow logic