Adding get reward to model