Add cost-benefit calculation functionality