Added the unit test