Use discounted reward rather than average