Added heuristics for AI