add in the rest of the tests