Also, add AI agent module