Adding unit test structure