Adding unit test