Adding unit test module