Change variable name in compute_cost