It should be more intuitive to change the blocks in the flow. Especially with code blocks where you actually need to change the code if the input changes. There should be like identifier of input blocks (like DataInput_1), that you use in the code instead of using the name of the previous stage (that might change).