working on mocking / testing