Updated Reinforcement logic