Update cost_generation_function.py