Using AI to augment humans