Add more tests to the tests