Adding unit-test