Add testing utilities