Adding a feedback to each stage