Adding unit test layout