custom_problem_template.py