Add more code to make workflow flow in diffrent directions according to user choice