Extended the model with Rewards