- Add cost function