Training for agents

Training for agents