Simplified the contract_benefit model and added unit tests