adding the tests a bit