added reinforcement learning