Update workflow.py