adding AI to this branch