AI.java updated