added the ability to put tasks into different stages