additional tweaking to AI