Add the source for the tests