Add heuristic function