and add the unit test