Add cost_function module