Update cost.py