Add get_problems call