Working on the unit test to check the calculations.