STAGE 5: REINFORCE