updated with RL agent