Fixing bug in task assignment code
Refining how the end-to-end pipline runs