Add test and modified methods reward

Add test usingReward and modified methods from class reward