Update HeuristicAI.java