changed my AI by adding test cases