updated to this new agent model