Altered AI slightly