Add get_problem function