Add the new tests here