Update pipeline.py