working on testing
 work on mocking