Add unit test framework