Small updates to agi benchmark