Tweaking the reward function