Add unit test module