Minor changes in the set reward method