Add unit-test