edited AI slightly