GENERATE_PYTHON_CODE_PROMPT = """
Generate a functional and correct Python code for the given problem.

Problem: """
MODIFY_CODE_PROMPT = """
Please improve or modify the following code to ensure it passes the test cases.

Code: """