Add some test stuff