add tests
making it pass the tests