Update future_cost.py code