Add unit-test framework