Fixed the reward function and the reward shown