Update create_problem_ids_collection.py