Add updated tests.