Modded the pipeline code