change class BaseAgent