Adding the unit test