Better heuristic function!