Fixed the AI's capability of learning