Adding unit test framework