Adds unit test stuff