adding unit test stuff