Update extra_cost.py