edit reward function