Adding an AI implementation