Add in more tests.