Added the new code for the new pipeline