Add new test functions