Update LearningAgent.py