Add reinforcement learning