Added support to workflow.yaml