Update problem.h

updated heuristic