working on the tests