Clean up AI a little